#include "PBDATA.H"



/****************************************
 蛇体 和 食物 相关的数据 和 函数
*****************************************/
uchar bodyRow[16], bodyCol[16];  // 蛇身的数组

uchar code dirRow[4] = {-1, 0, 1, 0};  // 上右下左方向X增量
uchar code dirCol[4] = {0, 1, 0, -1};  // 上右下左方向Y增量
extern uchar dir;
uchar maxRow = 8, maxCol = 8;  // 行的范围 [0, maxRow), 列范围同理
uchar speed = 5;  // 蛇的速度，单位是50ms
uchar snakeBodyLength = 0;  // 蛇身体的长度
uchar isDead = 0;  // 表示蛇是否死亡

uchar foodRow = 0, foodCol = 0, needCreate = 1;  // 食物坐标和食物是否被吃标志
 
void InitSnake()
{
	bodyRow[0] = 7;
	bodyCol[0] = 2;
	bodyRow[1] = 7;
	bodyCol[1] = 1;
	
	snakeBodyLength = 2;
	
	dir = 1;
	
	// 生成蛇打印的内容
	Matrix_GenerateTab(bodyRow, bodyCol, snakeBodyLength);
	
	// 设置时间种子
	srand(0);
}

void GenerateFood()
{
	uchar i = 0;
	while(needCreate) {
		// 随机生成 Row 和 Col
		foodRow = rand() % maxRow;
		foodCol = rand() % maxCol;
		
		// 判断食物是否和当前蛇身体冲突 
		for(i = 0; i < snakeBodyLength; i++) {
			if(bodyRow[i] == foodRow && bodyCol[i] == foodCol) {
				break;
			}
		}
		if(i == snakeBodyLength) {
			needCreate = 0;
			bodyRow[snakeBodyLength] = foodRow;
			bodyCol[snakeBodyLength] = foodCol;
		} 
	}
}

/****************************************
 定时器相关的代码
*****************************************/

uchar count;  // time = count * 50ms

void InitTimer()
{
  TMOD = 0x01;
  
  // 初始值 : 50ms
  TH0 = (65536 - 50000) / 256;  // 初始值取高八位
  TL0 = (65536 - 50000) % 256;  // 初始值取低八位
  
  // 中断开启
  ET0 = 1;  // 开启定时器0的中断
  EA = 1;   // 开启总的中断

  // 配置TCON
  // TR0 : 1, 启动定时器0
  TR0 = 1;    
}

void TimerIsr() interrupt 1
{
	uchar nextHeadRow = 0, nextHeadCol = 0, i = 0;
	
  // 重新装填
  // 初始值 : 50ms
  TH0 = (65536 - 50000) / 256;  // 初始值取高八位
  TL0 = (65536 - 50000) % 256;  // 初始值取低八位
  
  if(count == speed && !isDead) {  // count * 50ms 触发一次
    count = 0;
	
		// 生成食物
		GenerateFood();
		
		// 预测蛇头
		nextHeadRow = bodyRow[0] + dirRow[dir];
		nextHeadCol = bodyCol[0] + dirCol[dir];
		if(nextHeadRow >= maxRow || nextHeadRow < 0 
			 || nextHeadCol >= maxCol || nextHeadCol < 0) {
				isDead = 1;
				return;
		}
		
		// 身体撞击
		for(i = 0; i < snakeBodyLength; i++) {
			if(nextHeadRow == bodyRow[i] && nextHeadCol == bodyCol[i]) {
				isDead = 1;
				return;
			}
		}
		
		
		// 吃到食物与否
		if(nextHeadRow == foodRow && nextHeadCol == foodCol) {
			snakeBodyLength += 1;
			needCreate = 1;
			
			// 再生成新的食物
			GenerateFood();
		}
		
		// 蛇身体移动
		for(i = snakeBodyLength - 1; i > 0; i--) {
			bodyRow[i] = bodyRow[i - 1];
			bodyCol[i] = bodyCol[i - 1];
		}
		
		// 新的蛇头
		bodyRow[0] = nextHeadRow;
		bodyCol[0] = nextHeadCol;
		
		
		// 生成需要显示的图形
		Matrix_GenerateTab(bodyRow, bodyCol, snakeBodyLength + 1);
	}
	count++;	
	
	// 每50ms进行一次按键扫描
	DirKeyScan();
}

void main()
{
	uchar i = 0, a = 0;

	count = speed;
	InitSnake();
	InitTimer();

	while(1){
		if(isDead) {
			Matrix_ShowAll();
		} else {
			// 打印蛇的身体
			Matrix_ShowPointByCode();		
		}
	}
}